# 定义GPU数量变量
NUM_GPUS=2

#训练代码
CUDA_VISIBLE_DEVICES=2,3 python train_net_lmpm.py \
    --config-file configs/tlggnv1_SWIN_bs8.yaml \
    --num-gpus $NUM_GPUS --dist-url auto \
    --tlggn_k 20 \
    MODEL.WEIGHTS ./weights/model_final_86143f.pkl \
    OUTPUT_DIR ./outputs/tlggnv1/K20

sleep 15

#推理代码
CUDA_VISIBLE_DEVICES=0,1 python train_net_lmpm.py \
    --config-file configs/tlggnv1_SWIN_bs8.yaml \
    --num-gpus 2 --dist-url auto --eval-only \
    --tlggn_k 3 \
    MODEL.WEIGHTS ./outputs/tlggnv1/K3/model_final.pth \
    OUTPUT_DIR ./outputs/tlggnv1/K3


